package ontology;



import java.util.ArrayList;

public class InteractionTypeSet extends ArrayList<InteractionType>{




    public InteractionTypeSet getTypesWithHivProt(HivProt hivProt){
        InteractionTypeSet subset = new InteractionTypeSet();
        for(InteractionType t : this){
            if(t.getHivProt().equals(hivProt)){
                subset.add(t);
            }
        }
        return subset;
    }

    public InteractionTypeSet getTypesWithVerb(InteractionVerb verb){
        InteractionTypeSet subset = new InteractionTypeSet();
        for(InteractionType t : this){
            if(t.getVerb().equals(verb)){
                subset.add(t);
            }
        }
        return subset;
    }

    public InteractionVerbSet getVerbs(){
        InteractionVerbSet set = new InteractionVerbSet();
        for(InteractionType type : this){
            set.add(type.getVerb());
        }
        return set;
    }

}
